K Closest Points to Origin python